# -*- coding: utf-8 -*-

# Define your item pipelines here
#
# Don't forget to add your pipeline to the ITEM_PIPELINES setting
# See: https://docs.scrapy.org/en/latest/topics/item-pipeline.html
import pymysql


class ZhaobiaoPipeline(object):
    def __init__(self):
        self.mysql_conn = pymysql.Connection(
            host='localhost',
            port=3306,
            user='root',
            password='root',
            database='zhaobiao',
            charset='utf8',
        )

    def process_item(self, item, spider):
        # 创建光标对象
        cs = self.mysql_conn.cursor()
        # sql_str = 'insert into t_zhaobiao () value () '
        sql_column = ','.join([key for key in item.keys()])
        sql_value = ','.join(['"%s"' % item[key] for key in item.keys()])
        sql_str = 'insert into t_zhaobiao (%s) value (%s)' % (sql_column, sql_value)
        print('下载完成！', sql_str)
        cs.execute(sql_str)
        self.mysql_conn.commit()
        return item
